#!/usr/bin/env python
"""
WebSearchResearchAgent - 启动文件
直接运行此文件以启动代理
"""

import os
import sys

# 添加当前目录到路径，确保模块导入正确
current_dir = os.path.dirname(os.path.abspath(__file__))
if current_dir not in sys.path:
    sys.path.append(current_dir)

# 导入主模块
from main import run_agent

def main():
    """
    主入口函数
    """
    import argparse
    
    # 解析命令行参数
    parser = argparse.ArgumentParser(description="Web搜索研究代理")
    parser.add_argument("-q", "--question", help="要回答的问题")
    parser.add_argument("-s", "--search", help="初始搜索关键词")
    parser.add_argument("-d", "--debug", action="store_true", help="开启调试模式")
    args = parser.parse_args()
    
    # 运行代理
    result = run_agent(
        question=args.question,
        search_query=args.search,
        debug=args.debug
    )
    
    # 如果想要保存最终答案到文件
    if "answer" in result:
        with open("answer.txt", "w", encoding="utf-8") as f:
            f.write(result["answer"])
        print("✅ 答案已保存到 answer.txt")
    
    return result

if __name__ == "__main__":
    main() 